package com.example.utils;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import cn.hutool.json.JSONException;
import cn.hutool.json.JSONObject;
import cn.hutool.jwt.JWTPayload;
import cn.hutool.jwt.JWTUtil;
import cn.hutool.jwt.signers.JWTSigner;
import cn.hutool.jwt.signers.JWTSignerUtil;
import lombok.extern.slf4j.Slf4j;

import java.io.Serial;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Slf4j
public class JwtUtil {
	// 签发者
	private static final String ISSUER = "admin";
	// 定义一个密钥，用于签名
	private static final byte[] SECRET_KEY = "zeran".getBytes();
	// 定义生效时间(从当前时间的最开始)
	private static final Date NOT_BEFORE = DateUtil.beginOfSecond(DateUtil.date());
	// token过期时间
	private static final long EXPIRE_TIME = 1000 * 60 * 60 * 24 * 2;   // 2天
	// 签名工具
	private static final JWTSigner SIGNER = JWTSignerUtil.hs256(SECRET_KEY);

	/**
	 * 生成token
	 *
	 * @param userId 用户id
	 * @return jwt token
	 */
	public static String generateToken(String userId) {
		// 生成UUID
		String uuid = IdUtil.simpleUUID();
		// 创建JWT对象
		Map<String, Object> payload = new HashMap<>() {
			@Serial
			private static final long serialVersionUID = 1L;

			{
				put("iss", ISSUER); // 签发者
				put("uuid", uuid);  // 唯一标识
				put("uid", userId); // 用户id
				put(JWTPayload.NOT_BEFORE, NOT_BEFORE); // 生效时间
				put(JWTPayload.EXPIRES_AT, System.currentTimeMillis() + EXPIRE_TIME);   // 过期时间
			}
		};
		// 生成JWT Token
		return JWTUtil.createToken(payload, SIGNER);
	}

	/**
	 * 生成自定义过期时间的token
	 *
	 * @param userId 用户id
	 * @return jwt token
	 */
	public static String generateToken(String userId, long expireTime, TimeUnit timeUnit) {
		// 生成UUID
		String uuid = IdUtil.simpleUUID();
		// 创建JWT对象
		Map<String, Object> payload = new HashMap<>() {
			@Serial
			private static final long serialVersionUID = 1L;

			{
				put("iss", ISSUER); // 签发者
				put("uuid", uuid);  // 唯一标识
				put("uid", userId); // 用户id
				put(JWTPayload.NOT_BEFORE, NOT_BEFORE); // 生效时间
				put(JWTPayload.EXPIRES_AT, System.currentTimeMillis() + timeUnit.toMillis(expireTime));   // 过期时间
			}
		};
		// 生成JWT Token
		return JWTUtil.createToken(payload, SIGNER);
	}

	/**
	 * 解析token
	 *
	 * @param token token
	 * @return JSONObject对象包含payload信息
	 */
	public static JSONObject parseToken(String token) throws JSONException {
		// 解析JWT Token
		// 注意解析失败会抛出异常JSONException
		return JWTUtil.parseToken(token).getPayload().getClaimsJson();
	}

	/**
	 * 检查token是否有效
	 *
	 * @param token token
	 * @return 拆箱中要避免空指针异常
	 */
	public static Boolean verifyToken(String token) throws JSONException, NumberFormatException {
		// 1.验证JWT是否有效（hutool-jwt官方包有毛病）
		boolean isLegal = JWTUtil.verify(token, SIGNER);
		if (!isLegal) {
			return false;
		}
		long nowTime = System.currentTimeMillis();
		JSONObject entries = parseToken(token);
		// 2.手动校验是否过期
		String expS = entries.get(JWTPayload.EXPIRES_AT).toString();
		long expL = Long.parseLong(expS);
		if (nowTime > expL) {
			return false;
		}
		// 3.检验生效时间
		String nbfS = entries.get(JWTPayload.NOT_BEFORE).toString();
		long nbfL = Long.parseLong(nbfS);
		return nowTime >= nbfL;
	}

	/**
	 * 根据token解析出payload中的id
	 * @param token token
	 * @return userId
	 * @throws JSONException json解析异常
	 */
	public static String parseTokenId(String token) throws JSONException {
		JSONObject entries = parseToken(token);
		return entries.get("uid").toString();
	}

	public static void main(String[] args) {
		String token = generateToken("10001");
		System.out.println(token);
		System.out.println(verifyToken(token));
	}
}
